package com.cli.spiringbootlangchain4j.Controller;

import dev.langchain4j.model.openai.OpenAiChatModel;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class ChatController {

    static String apiKey = "sk-Z06jsByF6x7Xey2QYpQOTrjY9N5LaDoUxa7mDgymigGAXHhR";

    public static void main(String[] args) {
        OpenAiChatModel model = OpenAiChatModel.builder()
                .baseUrl("https://demo.voapi.top/v1")
                .apiKey(apiKey)
                .modelName("gpt-4")
                .build();

        String answer = model.chat("Say 'Hello World'");
        System.out.println(answer); // Hello World
    }


    @PostMapping("/chat")
    public Object chat(String userInput) {
        OpenAiChatModel model = OpenAiChatModel.builder()
                .apiKey(apiKey)
                .modelName("gpt-4o-mini")
                .build();

        // 这里可以调用 LangChain4j 的相关方法来处理用户输入
        // 例如，使用 LLM 模型生成回复
        // 返回一个简单的响应示例
        return "Chat response for input: " + userInput;

    }
}
